#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <regex>
#include <sys/stat.h>
#include "../server.hpp"

#define DEFALT_TIMEOUT 10

// _statu_msg 用于存储 HTTP 状态码与其描述的映射
std::unordered_map<int, std::string> _statu_msg = {
    {100, "Continue"},
    {101, "Switching Protocol"},
    {102, "Processing"},
    {103, "Early Hints"},
    {200, "OK"},
    {201, "Created"},
    {202, "Accepted"},
    {203, "Non-Authoritative Information"},
    {204, "No Content"},
    {205, "Reset Content"},
    {206, "Partial Content"},
    {207, "Multi-Status"},
    {208, "Already Reported"},
    {226, "IM Used"},
    {300, "Multiple Choice"},
    {301, "Moved Permanently"},
    {302, "Found"},
    {303, "See Other"},
    {304, "Not Modified"},
    {305, "Use Proxy"},
    {306, "unused"},
    {307, "Temporary Redirect"},
    {308, "Permanent Redirect"},
    {400, "Bad Request"},
    {401, "Unauthorized"},
    {402, "Payment Required"},
    {403, "Forbidden"},
    {404, "Not Found"},
    {405, "Method Not Allowed"},
    {406, "Not Acceptable"},
    {407, "Proxy Authentication Required"},
    {408, "Request Timeout"},
    {409, "Conflict"},
    {410, "Gone"},
    {411, "Length Required"},
    {412, "Precondition Failed"},
    {413, "Payload Too Large"},
    {414, "URI Too Long"},
    {415, "Unsupported Media Type"},
    {416, "Range Not Satisfiable"},
    {417, "Expectation Failed"},
    {418, "I'm a teapot"},
    {421, "Misdirected Request"},
    {422, "Unprocessable Entity"},
    {423, "Locked"},
    {424, "Failed Dependency"},
    {425, "Too Early"},
    {426, "Upgrade Required"},
    {428, "Precondition Required"},
    {429, "Too Many Requests"},
    {431, "Request Header Fields Too Large"},
    {451, "Unavailable For Legal Reasons"},
    {501, "Not Implemented"},
    {502, "Bad Gateway"},
    {503, "Service Unavailable"},
    {504, "Gateway Timeout"},
    {505, "HTTP Version Not Supported"},
    {506, "Variant Also Negotiates"},
    {507, "Insufficient Storage"},
    {508, "Loop Detected"},
    {510, "Not Extended"},
    {511, "Network Authentication Required"}};

// _mime_msg 用于存储文件扩展名与其 MIME 类型的映射
std::unordered_map<std::string, std::string> _mime_msg = {
    {".aac", "audio/aac"},
    {".abw", "application/x-abiword"},
    {".arc", "application/x-freearc"},
    {".avi", "video/x-msvideo"},
    {".azw", "application/vnd.amazon.ebook"},
    {".bin", "application/octet-stream"},
    {".bmp", "image/bmp"},
    {".bz", "application/x-bzip"},
    {".bz2", "application/x-bzip2"},
    {".csh", "application/x-csh"},
    {".css", "text/css"},
    {".csv", "text/csv"},
    {".doc", "application/msword"},
    {".docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    {".eot", "application/vnd.ms-fontobject"},
    {".epub", "application/epub+zip"},
    {".gif", "image/gif"},
    {".htm", "text/html"},
    {".html", "text/html"},
    {".ico", "image/vnd.microsoft.icon"},
    {".ics", "text/calendar"},
    {".jar", "application/java-archive"},
    {".jpeg", "image/jpeg"},
    {".jpg", "image/jpeg"},
    {".js", "text/javascript"},
    {".json", "application/json"},
    {".jsonld", "application/ld+json"},
    {".mid", "audio/midi"},
    {".midi", "audio/x-midi"},
    {".mjs", "text/javascript"},
    {".mp3", "audio/mpeg"},
    {".mpeg", "video/mpeg"},
    {".mpkg", "application/vnd.apple.installer+xml"},
    {".odp", "application/vnd.oasis.opendocument.presentation"},
    {".ods", "application/vnd.oasis.opendocument.spreadsheet"},
    {".odt", "application/vnd.oasis.opendocument.text"},
    {".oga", "audio/ogg"},
    {".ogv", "video/ogg"},
    {".ogx", "application/ogg"},
    {".otf", "font/otf"},
    {".png", "image/png"},
    {".pdf", "application/pdf"},
    {".ppt", "application/vnd.ms-powerpoint"},
    {".pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    {".rar", "application/x-rar-compressed"},
    {".rtf", "application/rtf"},
    {".sh", "application/x-sh"},
    {".svg", "image/svg+xml"},
    {".swf", "application/x-shockwave-flash"},
    {".tar", "application/x-tar"},
    {".tif", "image/tiff"},
    {".tiff", "image/tiff"},
    {".ttf", "font/ttf"},
    {".txt", "text/plain"},
    {".vsd", "application/vnd.visio"},
    {".wav", "audio/wav"},
    {".weba", "audio/webm"},
    {".webm", "video/webm"},
    {".webp", "image/webp"},
    {".woff", "font/woff"},
    {".woff2", "font/woff2"},
    {".xhtml", "application/xhtml+xml"},
    {".xls", "application/vnd.ms-excel"},
    {".xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    {".xml", "application/xml"},
    {".xul", "application/vnd.mozilla.xul+xml"},
    {".zip", "application/zip"},
    {".3gp", "video/3gpp"},
    {".3g2", "video/3gpp2"},
    {".7z", "application/x-7z-compressed"}};

// 在 HTTP 请求/响应中：当服务器返回一个文件时，它会在 HTTP 响应头中包含一个 Content-Type 字段，这个字段的值就是 MIME 类型。这样，客户端可以根据 MIME 类型来确定如何处理这个文件（例如：显示图片、播放视频、渲染 HTML、下载文件等）。
// 确保正确处理文件：通过 MIME 类型，浏览器能够正确地处理不同类型的文件。例如：
// .html 文件的 MIME 类型是 text/html，浏览器会将其作为 HTML 文件来解析和渲染。
// .jpg 文件的 MIME 类型是 image/jpeg，浏览器会将其作为图片来显示。

class Util
{
public:
    // 字符串分割函数,将src字符串按照sep字符进行分割，得到的各个字串放到arry中，最终返回字串的数量
    static size_t Split(const std::string &src, const std::string &sep, std::vector<std::string> *arry)
    {
        size_t offset = 0;
        // 这是一个循环，它会继续执行直到 offset 超过字符串 src 的大小。也就是说，循环会遍历整个字符串，查找分隔符并提取子字符串。
        // 有10个字符，offset是查找的起始位置，范围应该是0~9，offset==10就代表已经越界了
        while (offset < src.size())
        {
            // 在src字符串偏移量offset处，开始向后查找sep字符/字串，返回查找到的位置
            size_t pos = src.find(sep, offset);
            if (pos == std::string::npos)
            {
                // 没有找到特定的字符
                // 将剩余的部分当作一个字串，放入arry中
                // 如果此时 pos 等于 src.size()，则意味着已经到了字符串的结尾，跳出循环。
                if (pos == src.size())
                    break;
                // substr 从 offset 开始直接截取到最后
                arry->push_back(src.substr(offset));
                return arry->size();
            }
            // 如果找到的分隔符位于当前 offset 位置（即字符串中有连续的分隔符），则意味着当前的子字符串是一个空字符串（没有内容）
            if (pos == offset)
            {
                // 将 offset 更新为 pos + sep.size()，即跳过当前的分隔符，继续查找下一个子字符串
                offset = pos + sep.size();
                // 跳过当前循环，进入下一次循环的开始
                continue;
            }
            arry->push_back(src.substr(offset, pos - offset));
            offset = pos + sep.size();
        }
        return arry->size();
    }

    // 读取文件的所有内容，将读取的内容放到一个Buffer中
    static bool ReadFile(const std::string &filename, std::string *buf)
    {
        // 使用 std::ifstream 来打开文件，std::ios::binary 表示以二进制模式打开文件（可以避免文本模式下的行结束符问题）
        std::ifstream ifs(filename, std::ios::binary);
        if (ifs.is_open() == false)
        {
            printf("OPEN %s FILE FAILED!!", filename.c_str());
            return false;
        }
        // seekg(0, ifs.end)：将读写指针移到文件末尾。
        // tellg()：获取当前位置的偏移量，此时的位置是文件的末尾，因此返回的值就是文件的大小（单位是字节）。
        size_t fsize = 0;
        ifs.seekg(0, ifs.end); // 跳转读写位置到末尾
        fsize = ifs.tellg();   // 获取当前读写位置相对于起始位置的偏移量，从末尾偏移刚好就是文件大小
        // 重新将文件指针移回文件开头
        ifs.seekg(0, ifs.beg); // 跳转到起始位置
        // 使用 resize(fsize) 调整 buf 的大小，确保它有足够的空间来存储文件的内容。
        buf->resize(fsize); // 开辟文件大小的空间
        // 将文件内容从文件指针当前位置读取到 buf 中。这里的 &(*buf)[0] 是获取 std::string 内部字符数组的指针，作为 read 函数的目标缓冲区
        ifs.read(&(*buf)[0], fsize);
        // ifs.good() 检查流的状态，返回 true 表示文件读取成功，返回 false 表示读取失败。如果读取失败，输出错误信息并关闭文件流，返回 false。
        if (ifs.good() == false)
        {
            printf("READ %s FILE FAILED!!", filename.c_str());
            ifs.close();
            return false;
        }
        ifs.close();
        return true;
    }

    // 向文件写入数据
    static bool WriteFile(const std::string &filename, const std::string &buf)
    {
        // std::ios::trunc 表示在文件打开时会截断文件，即清空原文件的内容（如果文件已存在），然后重新写入。
        std::ofstream ofs(filename, std::ios::binary | std::ios::trunc);
        if (ofs.is_open() == false)
        {
            printf("OPEN %s FILE FAILED!!", filename.c_str());
            return false;
        }
        // ofs.write() 将 buf 中的数据写入到文件中，使用 c_str() 获取数据的地址，并写入指定大小的数据。
        ofs.write(buf.c_str(), buf.size());
        if (ofs.good() == false)
        {
            ERR_LOG("WRITE %s FILE FAILED!", filename.c_str());
            ofs.close();
            return false;
        }
        ofs.close();
        return true;
    }

    // URL 编码 是为了将特殊字符、非 ASCII 字符和空格等转换为 URL 安全格式，避免它们与 URL 结构发生冲突，并确保信息能够正确传输。
    // URL 解码 则是将这些编码后的字符恢复为原始数据，使服务器或客户端能够正确处理和使用这些信息。
    // URL编码，避免URL中资源路径与查询字符串中的特殊字符与HTTP请求中特殊字符产生歧义
    // 编码格式：将特殊字符的ascii值，转换为两个16进制字符，前缀%   C++ -> C%2B%2B
    //   不编码的特殊字符： RFC3986文档规定 . - _ ~ 字母，数字属于绝对不编码字符
    // RFC3986文档规定，编码格式 %HH
    // W3C标准中规定，查询字符串中的空格，需要编码为+， 解码则是+转空格
    // UrlEncode 是一个 URL 编码函数。它的作用是将一个字符串 url 中的特殊字符转换成符合 URL 编码标准的形式，通常用于将 URL 中的非标准字符（如空格、特殊符号等）转换成 %HH 格式，其中 HH 是字符的 ASCII 十六进制表示。
    static std::string UrlEncode(const std::string url, bool convert_space_to_plus)
    {
        std::string res;
        for (auto &c : url)
        {
            // 这一段是判断字符是否是符合 URL 标准的安全字符，符合的字符将直接加入到结果字符串 res 中，而不进行编码。
            // 符合条件的字符包括：
            // 字符 .（句点）
            // 字符 -（连字符）
            // 字符 _（下划线）
            // 字符 ~（波浪号）
            // 字符集 isalnum(c)，即字母和数字字符（a-zA-Z0-9）
            // 这些字符是 URL 中合法的字符，可以直接保留在 URL 中，不需要编码。
            if (c == '.' || c == '-' || c == '_' || c == '~' || isalnum(c))
            {
                res += c;
                continue;
            }
            // 在 URL 编码中，空格通常用加号（+）表示，尤其是在查询字符串中。设置 convert_space_to_plus 为 true 时，空格就会被替换为 +。
            if (c == ' ' && convert_space_to_plus == true)
            {
                res += '+';
                continue;
            }
            // 剩下的字符都是需要编码成为 %HH 格式
            // tmp 是一个 char 数组，大小为 4，意味着它能够容纳最多 3 个十六进制字符（如 %21）和一个结尾的 \0 字符，确保字符串是以 \0 结尾的有效字符串。
            char tmp[4] = {0};
            // snprintf 与 printf比较类似，都是格式化字符串，只不过一个是打印，一个是放到一块空间中
            snprintf(tmp, 4, "%%%02X", c);
            res += tmp;
        }
        // 例如，调用 UrlEncode("Hello World!") 时，假设 convert_space_to_plus 为 true，返回值将是："Hello+World%21"（%21 是 ! 的 ASCII 十六进制表示）。
        return res;
    }

    static char HEXTOI(char c)
    {
        if (c >= '0' && c <= '9')
        {
            return c - '0';
        }
        else if (c >= 'a' && c <= 'z')
        {
            return c - 'a' + 10;
        }
        else if (c >= 'A' && c <= 'Z')
        {
            return c - 'A' + 10;
        }
        // 如果字符 c 既不是数字字符（'0'-'9'），也不是小写字母字符（'a'-'f'），或者大写字母字符（'A'-'F'），函数会返回 -1
        // 这表示输入字符不是合法的十六进制字符。在十六进制表示中，只能有 0-9、a-f 或 A-F 这 16 个字符，其他字符应被视为无效
        return -1;
    }

    static std::string UrlDecode(const std::string url, bool convert_plus_to_space)
    {
        // 遇到了%，则将紧随其后的2个字符，转换为数字，第一个数字左移4位，然后加上第二个数字  + -> 2b  %2b->2 << 4 + 11
        // res 用来存储最终的解码结果字符串。在解码过程中，所有转换后的字符都会被添加到这个字符串中。
        std::string res;
        for (int i = 0; i < url.size(); i++)
        {
            if (url[i] == '+' && convert_plus_to_space == true)
            {
                res += ' ';
                continue;
            }
            // 如果当前字符是 %，表示接下来的两个字符是十六进制编码，代表一个字符的 ASCII 值。
            // (i + 2) < url.size() 这个条件保证了当前字符 i 后面至少有两个字符，以确保能够安全地访问 % 后面的两个字符。
            if (url[i] == '%' && (i + 2) < url.size())
            {
                // HEXTOI(url[i + 1]) 和 HEXTOI(url[i + 2]) 将 % 后面的两个字符（url[i + 1] 和 url[i + 2]）分别转换为十六进制的数字值。
                char v1 = HEXTOI(url[i + 1]);
                char v2 = HEXTOI(url[i + 2]);
                // v1 * 16 + v2 将两个十六进制数字组合成一个字节。例如，如果 v1 = 2，v2 = 11，那么 v 将是 2 * 16 + 11 = 43。
                char v = v1 * 16 + v2;
                res += v;
                // i += 2 跳过接下来的两个字符（即 % 后的两个字符），因为它们已经被解码并处理了。
                i += 2;
                continue;
            }
            res += url[i];
        }
        return res;
    }

    // 响应状态码的描述信息获取
    static std::string StatuDesc(int statu)
    {

        auto it = _statu_msg.find(statu);
        if (it != _statu_msg.end())
        {
            return it->second;
        }
        return "Unknow";
    }

    // 根据文件后缀名获取文件mime
    static std::string ExtMime(const std::string &filename)
    {

        // a.b.txt  先获取文件扩展名
        size_t pos = filename.find_last_of('.');
        if (pos == std::string::npos)
        {
            // 如果没有找到点（即没有扩展名），返回默认的 MIME 类型 "application/octet-stream"。
            return "application/octet-stream";
        }
        // 根据扩展名，获取mime
        std::string ext = filename.substr(pos);
        auto it = _mime_msg.find(ext);
        if (it == _mime_msg.end())
        {
            return "application/octet-stream";
        }
        return it->second;
    }

    // 判断一个文件是否是一个目录
    static bool IsDirectory(const std::string &filename)
    {
        // 这里声明了一个 stat 结构体变量 st。stat 是一个结构体，用来存储文件的各种属性信息（如文件类型、大小、权限等）。它在 <sys/stat.h> 中定义。
        // st_mode: 文件的类型和权限信息。
        // st_size: 文件的大小。
        // st_uid: 文件的所有者。
        // st_gid: 文件的所属组。
        struct stat st;
        // stat 是一个系统调用，作用是获取文件或目录的元数据（如文件类型、大小、权限等）。
        // stat 函数返回 0 表示成功，返回负数表示失败（如文件不存在或没有访问权限）。
        int ret = stat(filename.c_str(), &st);
        if (ret < 0)
        {
            return false;
        }
        // S_ISDIR(st.st_mode)：如果 st.st_mode 表示该文件是一个目录，返回 true，否则返回 false。
        return S_ISDIR(st.st_mode);
    }

    // 判断一个文件是否是一个普通文件
    static bool IsRegular(const std::string &filename)
    {
        struct stat st;

        int ret = stat(filename.c_str(), &st);
        if (ret < 0)
        {
            return false;
        }
        return S_ISREG(st.st_mode);
    }

    // http请求的资源路径有效性判断
    //  /index.html  --- 前边的/叫做相对根目录  映射的是某个服务器上的子目录
    //  想表达的意思就是，客户端只能请求相对根目录中的资源，其他地方的资源都不予理会
    //  /../login, 这个路径中的..会让路径的查找跑到相对根目录之外，这是不合理的，不安全的
    static bool ValidPath(const std::string &path)
    {
        // 思想：按照/进行路径分割，根据有多少子目录，计算目录深度，有多少层，深度不能小于0
        std::vector<std::string> subdir;
        Split(path, "/", &subdir);
        // 变量 level 用于跟踪当前的目录深度。初始值为 0，表示根目录。
        int level = 0;
        for (auto &dir : subdir)
        {
            // ".."：回到父目录，深度减一。
            // 普通目录名：进入下一级目录，深度加一。
            if (dir == "..")
            {
                level--; // 任意一层走出相对根目录，就认为有问题
                if (level < 0)
                    return false;
                continue;
            }
            level++;
        }
        return true;
    }

    class HttpRequest
    {
    public:
        std::string _method;                                   // 请求方法
        std::string _path;                                     // 资源路径
        std::string _version;                                  // 协议版本
        std::string _body;                                     // 请求正文
        std::smatch _matches;                                  // 资源路径的正则提取数据
        std::unordered_map<std::string, std::string> _headers; // 头部字段
        std::unordered_map<std::string, std::string> _params;  // 查询字符串
    public:
        HttpRequest() : _version("HTTP/1.1") {}
        void ReSet()
        {
            _method.clear();
            _path.clear();
            _version = "HTTP/1.1";
            _body.clear();
            std::smatch match;
            _matches.swap(match);
            _headers.clear();
            _params.clear();
        }

        // 插入头部字段
        void SetHeader(const std::string &key, const std::string &val)
        {
            _headers.insert(std::make_pair(key, val));
        }

        // 判断是否存在指定头部字段
        bool HasHeader(const std::string &key) const
        {
            auto it = _headers.find(key);
            if (it == _headers.end())
            {
                return false;
            }
            return true;
        }

        // 获取指定头部字段的值
        std::string GetHeader(const std::string &key) const
        {
            auto it = _headers.find(key);
            if (it == _headers.end())
            {
                return "";
            }
            return it->second;
        }

        // 插入查询字符串
        void SetParam(const std::string &key, const std::string &val)
        {
            _params.insert(std::make_pair(key, val));
        }

        // 判断是否有某个指定的查询字符串
        bool HasParam(const std::string &key) const
        {
            auto it = _params.find(key);
            if (it == _params.end())
            {
                return false;
            }
            return true;
        }

        // 获取指定的查询字符串
        std::string GetParam(const std::string &key) const
        {
            auto it = _params.find(key);
            if (it == _params.end())
            {
                return "";
            }
            return it->second;
        }

        // 获取正文长度
        size_t ContentLength() const
        {
            // Content-Length: 1234\r\n
            bool ret = HasHeader("Content-Length");
            if (ret == false)
            {
                return 0;
            }
            std::string clen = GetHeader("Content-Length");
            return std::stol(clen);
        }

        // 判断是否是短链接
        bool Close() const
        {
            // 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
            if (HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
            {
                return false;
            }
            return true;
        }
    };

    class HttpResponse
    {
    public:
        int _statu; // HTTP 响应的状态码
        bool _redirect_flag; // 标记是否需要重定向
        // HTTP中的重定向是指当客户端请求的资源（如网页、图片等）已经被移动到另一个位置时，服务器会通过返回特定的状态码和新的 URL 告诉客户端重新发起请求到新的地址。重定向通常用于处理资源的移动、更新或临时改变位置。
        std::string _body; // 响应的主体内容
        std::string _redirect_url; // 如果响应需要重定向，该成员保存重定向的目标 URL
        std::unordered_map<std::string, std::string> _headers; // 用于存储 HTTP 响应头。每个头部字段及其对应的值都保存在这个映射中

    public:
        HttpResponse() : _redirect_flag(false), _statu(200) {}
        HttpResponse(int statu) : _redirect_flag(false), _statu(statu) {}
        void ReSet()
        {
            _statu = 200;
            _redirect_flag = false;
            _body.clear();
            _redirect_url.clear();
            _headers.clear();
        }

        // 插入头部字段
        void SetHeader(const std::string &key, const std::string &val)
        {
            _headers.insert(std::make_pair(key, val));
        }

        // 判断是否存在指定头部字段
        bool HasHeader(const std::string &key)
        {
            auto it = _headers.find(key);
            if (it == _headers.end())
            {
                return false;
            }
            return true;
        }

        // 获取指定头部字段的值
        std::string GetHeader(const std::string &key)
        {
            auto it = _headers.find(key);
            if (it == _headers.end())
            {
                return "";
            }
            return it->second;
        }

        void SetContent(const std::string &body, const std::string &type = "text/html")
        {
            _body = body;
            SetHeader("Content-Type", type);
        }

        void SetRedirect(const std::string &url, int statu = 302)
        {
            _statu = statu;
            _redirect_flag = true;
            _redirect_url = url;
        }

        // 判断是否是短链接
        bool Close()
        {
            // 没有Connection字段，或者有Connection但是值是close，则都是短链接，否则就是长连接
            if (HasHeader("Connection") == true && GetHeader("Connection") == "keep-alive")
            {
                return false;
            }
            return true;
        }
    };

    
};